package config

import (
	"demos/model"
	"fmt"
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
)

// DB 数据库连接
var DB *gorm.DB

// InitDB 初始化数据库
func InitDB() {
	dsn := "root:111111@tcp(127.0.0.1:3306)/demo?charset=utf8mb4&parseTime=True&loc=Local"
	db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
	if err != nil {
		panic("failed to connect database: " + err.Error())
	}
	DB = db
	// 自动迁移用户表和Post模型
	err = DB.AutoMigrate(&model.User{}, &model.Post{}, &model.PostView{}, &model.Comment{})
	if err != nil {
		return
	}
	fmt.Println("Database connected.")
}
